/*
 * $Id: MyOpenIDAuthenticationFilter.java 42 2013-04-17 15:33:59Z gabakyan $
 * $Author: gabakyan $
 * $Revision: 42 $
 * $Date: 2013-04-17 15:33:59 +0000 (Wed, 17 Apr 2013) $
 *
 * Copyright (c) 2013 Supply Chain Intelligence (SCI), Inc.
 * http://www.scintelligence.com/, Email: info@scintelligence.com
 * All rights reserved.
 *
 * This file is part of Logistics Map.
 *
 * Logistics Map is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, version 3 of the License.
 *
 * Logistics Map is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Logistics Map.  If not, see <http://www.gnu.org/licenses/>.
 */

package com.sci.logisticsmap.web.security.filter;

import com.sci.logisticsmap.web.security.support.SecurityConstants;
import com.sci.logisticsmap.web.security.support.SecurityUtilities;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.security.openid.OpenIDAuthenticationFilter;

import javax.servlet.http.HttpServletRequest;

/**
 * The class {@code MyOpenIDAuthenticationFilter} represents the OpenID authentication custom filter.
 *
 * @author gabakyan@gmail.com (Last modified by $Author: gabakyan $)
 * @version $Revision: 42 $
 */
public class MyOpenIDAuthenticationFilter extends OpenIDAuthenticationFilter {
    /**
     * The logger.
     */
    private static final Logger logger;

    /**
     * The app url property.
     */
    private @Value("${app.url}") String appUrl;

    /**
     * The static initialization block.
     */
    static {
        logger = LoggerFactory.getLogger(MyOpenIDAuthenticationFilter.class);
    }

    /*
     * (non-Javadoc)
     *
     * @see OpenIDAuthenticationFilter#buildReturnToUrl(javax.servlet.http.HttpServletRequest)
     */
    @Override
    protected String buildReturnToUrl(HttpServletRequest request) {
        String url = super.buildReturnToUrl(request);

        String pathAndQueryString = SecurityUtilities.getPathAndQueryString(url);
        //logger.debug("buildReturnToUrl: url path and query string = {}", pathAndQueryString);
        if (pathAndQueryString.contains(SecurityConstants.OPENID_LOGIN_PROCESSING_URL)
                || pathAndQueryString.contains(SecurityConstants.ADMIN_OPENID_LOGIN_PROCESSING_URL)) {
            String serverAndPort = SecurityUtilities.getServerAndPort(url);
            logger.debug("buildReturnToUrl: url server and port = {}", serverAndPort);

            String appUrlServerAndPort = SecurityUtilities.getServerAndPort(appUrl);
            logger.debug("buildReturnToUrl: appUrl server and port = {}", appUrlServerAndPort);

            if (!serverAndPort.equals(appUrlServerAndPort)) {
                String newUrl = SecurityUtilities.replaceServerAndPort(url, appUrlServerAndPort);
                logger.debug("buildReturnToUrl: new url = {}", newUrl);
                return url;
            }
        }

        return url;
    }
}
